package com.nowcoder.topic.dfs.hard;

/**
 * NC6 二叉树中的最大路径和
 * @author d3y1
 */
public class NC6{
    private static int result = Integer.MIN_VALUE;

    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 程序入口
     *
     * @param root TreeNode类
     * @return int整型
     */
    public int maxPathSum (TreeNode root) {
        postorder(root);

        return result;
    }

    /**
     * 递归: 后序遍历
     * @param root
     * @return
     */
    private int postorder(TreeNode root){
        if(root == null){
            return 0;
        }

        // 左分支最大路径值
        int leftMaxVal = postorder(root.left);
        // 右分支最大路径值
        int rightMaxVal = postorder(root.right);
        // 当前节点处最大路径和 当前节点值+左分支最大路径值+右分支最大路径值
        int currSum = root.val;
        if(leftMaxVal > 0){
            currSum += leftMaxVal;
        }
        if(rightMaxVal > 0){
            currSum += rightMaxVal;
        }
        result = Math.max(result, currSum);

        // 当前节点处最大路径值 左分支或右分支或仅取当前节点值
        int currVal = root.val;
        int max = Math.max(leftMaxVal, rightMaxVal);
        if(max > 0){
            currVal += max;
        }

        return currVal;
    }

    private class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;
        }
    }
}